tensor([[0.3800, 0.3800, 0.3800, 0.3800, 0.3800],
        [0.6200, 0.6200, 0.6200, 0.6200, 0.6200]], grad_fn=<RoundBackward1>)